import os
import shutil

output_path = './output'

for file_name in os.listdir(output_path):
    if file_name == '.gitkeep':
        continue
    file_path = os.path.join(output_path, file_name)
    if os.path.isfile(file_path) or os.path.islink(file_path):
        os.unlink(file_path)
    elif os.path.isdir(file_path):
        shutil.rmtree(file_path)